package com.orm.mybatisplus.config;

/**
 * 自定义数据源切换类
 */
public class DatabaseContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    /**
     * 设置数据源
     */
    public static void setDBKey(String dataSourceKey) {
        contextHolder.set(dataSourceKey);
    }

    /**
     * 取得当前数据源
     */
    public static String getDBKey() {
        return contextHolder.get();
    }

    /**
     * 清除上下文数据
     */
    public static void clearDBKey() {
        contextHolder.remove();
    }
}
